# memory/__init__.py
# 增强记忆系统模块

from .local_embedding_client import LocalEmbeddingClient
from .memory_manager import EnhancedMemoryManager, MemoryType, MemoryImportance
from .rag_retriever import EnhancedRAGRetriever
from .rag_manager import (
    EnhancedGraphRAGManager,
    initialize_enhanced_graph_rag, 
    get_enhanced_rag_manager, 
    close_enhanced_graph_rag
)

__all__ = [
    'LocalEmbeddingClient', 
    'EnhancedMemoryManager', 'MemoryType', 'MemoryImportance',
    'EnhancedRAGRetriever', 'EnhancedGraphRAGManager',
    'initialize_enhanced_graph_rag', 'get_enhanced_rag_manager', 'close_enhanced_graph_rag'
]
